from langchain_core.prompts import PromptTemplate
from langchain_openai import ChatOpenAI
import os
from dotenv import load_dotenv
from pydantic import SecretStr


# 加载.env文件中的环境变量
load_dotenv()

llm = ChatOpenAI(
    model="qwen-flash",
    base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    api_key=SecretStr(os.getenv("DASHSCOPE_API_KEY")),
    streaming=True
)

# 创建提示词模板
prompt_template = PromptTemplate.from_template("你做的{something}真棒!")
# 模板 -> 提示词
prompt = prompt_template.format(something="程序")
print(prompt)
# 输入给AI模型, 获取结果
resp = llm.invoke(prompt)
print(resp.content)
